

class Tools:
    def __init__(self):
         # 定义需要高亮的关键词及其对应颜色
        self.highlight_words = {
            "Reasoning": "blue",
            "Question": "red", 
            "Answer": "green",
            "Total time": "magenta"
        }
        
    # 定义文本高亮函数
    from IPython.display import display,Markdown
    def colorize_text(self, text):
        """
        对文本中的特定关键词进行颜色高亮处理
        
        参数:
        text (str): 需要处理的原始文本
        
        返回:
        str: 带有HTML颜色标签的格式化文本
        """
        # 对每个关键词进行颜色高亮处理
        for word, color in self.highlight_words.items():
            # 替换关键词，添加HTML颜色标签和换行
            text = text.replace(
                f"{word}:", 
                f"\n\n**<font color='{color}'>{word}:</font>**"
            )
        return text